var axios = require('axios');

function 生成育儿文章(content) {
    var dpsk_api_key = "sk-906c9c52e47f412f8067a5a148e6956b";
    var dpsk_base_url = "https://api.deepseek.com/v1/chat/completions";

    var headers = {
        "Authorization": "Bearer " + dpsk_api_key,
        "Content-Type": "application/json"
    };

    var messages = [
        // { "role": "system", "content": "You are a helpful assistant." },
        { "role": "user", "content": "以孤独的夜行者为题写一篇750字的散文，描绘一个人在城市中夜晚漫无目的行走的心情与所见所感，以及夜的寂静给予的独特感悟。" }
    ];

    var resultText = ""; // 存储完整的返回数据

    function 请求API() {
        var data = {
            "model": "deepseek-chat",
            "messages": messages,
            "stream": true,
            "max_tokens": 1000,
            "temperature": 1.5
        };

        axios.post(dpsk_base_url, data, {
            headers: headers,
            responseType: 'stream'
        }).then(function(response) {


            response.data.on('data', function(chunk) {
                var text = chunk.toString()
                if (text.startsWith("data:")) {
                    try {
                        var jsonData = JSON.parse(text.substring(5)); // 解析 JSON
                        if (jsonData.choices && jsonData.choices[0].delta.content) {
                            var contentPart = jsonData.choices[0].delta.content;
                            console.log(contentPart); // 实时输出解析出的内容
                            resultText += contentPart; // 累加数据
                        }
                    } catch (e) {
                        console.error("JSON 解析失败：", e.message);
                    }
                }
            });

            response.data.on('end', function() {
                console.log("\nAPI 调用完成。");
                console.log("最终返回的数据：", resultText); // 输出完整数据
                
                // 如果文章长度不够，继续请求扩展内容
                if (resultText.length < 1000) {
                    console.log("文章未达到 1000 字，继续请求补充内容...");
                    messages.push({ "role": "user", "content": "请继续扩展关于'" + content + "'的文章，使内容更加详细、完整，并保持与原主题一致。" });
                    请求API();
                }
            });
        }).catch(function(error) {
            console.error("API 调用失败：", error.response ? error.response.data : error.message);
        });
    }

    请求API();
}

生成育儿文章('如何培养孩子的阅读习惯');
